################### transformer-code
# transformer-code-train:
CUDA_VISIBLE_DEVICES=4 python -m tasks.ClassSummary.models.mTransformer.main \
  --save_path ./model/transformer-code/ \
  --input_flag code \
  --train ./tasks/ClassSummary/resource/train.token.jsonl \
  --val ./tasks/ClassSummary/resource/valid.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/code.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --patience 4 \
  --batch_size 32 \
  --max_epoch 100 \
  --embedding 1024 \
  --dffn 2048 \
  --layers 3 \
  --nhead 4 \
  --lr 0.0001 \
  --max_grad 5 \
  --gpu_num 1 \
  --seed 1024 \
  --check_freq 1

# transformer-code-hybrid-test:
CUDA_VISIBLE_DEVICES=7 python -m tasks.ClassSummary.models.mTransformer.test \
  --save_path ./model/transformer-code-hybrid.jsonl \
  --input_flag code \
  --test ./tasks/ClassSummary/resource/test.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/code.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/transformer-code/Transformer-code-epoch=18-bleu=4.49.ckpt \
  --save_path ./translate/transformer-code-hybrid.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024

# transformer--code-cross-test:
CUDA_VISIBLE_DEVICES=7 python -m tasks.ClassSummary.models.mTransformer.test \
  --save_path ./model/transformer-code-cross.jsonl \
  --input_flag code \
  --test ./tasks/ClassSummary/resource/cross.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/code.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/transformer-code/Transformer-code-epoch=18-bleu=4.49.ckpt \
  --save_path ./translate/transformer-code-cross.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024
